# Dataset configuration

# 数据集参数配置
dataset_config = {
    'DroneDetectionDataset': {
        # dataset
        'data_root': r'G:\pycharmProjs\anti-uav-metrics\spatial_temporal\dataset\Drone-detection-dataset',
        'gt_folder': './evaluator/groundtruths_ddd/',  # 存放测试集 视频帧 标注文件

        # input size
        'train_size': 224,
        'test_size': 224,

        # transform
        'pixel_mean': [0., 0., 0.],
        'pixel_std': [1., 1., 1.],
        'jitter': 0.2,
        'hue': 0.1,
        'saturation': 1.5,
        'exposure': 1.5,
        'sampling_rate': 1,
        'len_clip': 16,

        # cls label
        'multi_hot': False,  # one hot

        # post process
        'conf_thresh': 0.3,
        'nms_thresh': 0.5,
        'conf_thresh_val': 0.005,
        'nms_thresh_val': 0.5,

        # freeze backbone
        'freeze_backbone_2d': False,
        'freeze_backbone_3d': False,

        # train config
        'batch_size': 1,
        'test_batch_size': 8,
        'accumulate': 16,
        'optimizer': 'adamw',
        'momentum': 0.9,
        'weight_decay': 5e-4,
        'max_epoch': 5,
        'lr_epoch': [1, 2, 3, 4],
        'base_lr': 1e-4,
        'lr_decay_ratio': 0.5,

        # warmup strategy
        'warmup': 'linear',
        'warmup_factor': 0.00066667,
        'wp_iter': 500,

        # class names
        'valid_num_classes': 4,
        'label_map': (
            "AIRPLANE", "BIRD", "DRONE", "HELICOPTER"
        ),
    },
}
